package com.husong.common.shiro;

import org.apache.shiro.SecurityUtils;
import org.apache.shiro.authc.AuthenticationException;
import org.apache.shiro.authc.AuthenticationInfo;
import org.apache.shiro.authc.AuthenticationToken;
import org.apache.shiro.authz.AuthorizationInfo;
import org.apache.shiro.authz.SimpleAuthorizationInfo;
import org.apache.shiro.realm.AuthorizingRealm;
import org.apache.shiro.session.Session;
import org.apache.shiro.subject.PrincipalCollection;
import org.apache.shiro.subject.Subject;

/**
 * @author MaxMin.
 * @email tjljob@126.com
 * @date 2015/8/26 23:20.
 */
public class AuthorizRealm extends AuthorizingRealm {

//	@Autowired
//	ManagerService managerService;
//	
//	@Autowired
//	UserRoleService userRoleService;
//	
//	@Autowired
//	PermissionService permissionService;
//	
//	@Autowired
//	RoleService roleService;

	/**
	 * 为当前登录的Subject授予角色和权限 该方法的调用时机为需授权资源被访问时
	 * 如果未启用AuthorizationCache，每次访问需授权资源时都会执行该方法中的逻辑
	 * 使用Spring3.1开始提供的ConcurrentMapCache支持,则可灵活决定是否启用AuthorizationCache
	 * 如果从数据库获取权限信息时,先去访问Spring3.1提供的缓存,而不使用Shior提供的AuthorizationCache
	 */
	@Override
	protected AuthorizationInfo doGetAuthorizationInfo(
			PrincipalCollection principals) {
//		// 获取当前登录的用户名,等价于(String)principals.fromRealm(this.getName()).iterator().next()
//		String currentUsername = (String) super
//				.getAvailablePrincipal(principals);
//		List<String> roleList = new ArrayList<String>();
//		List<String> permissionList = new ArrayList<String>();
//		// //从数据库中获取当前登录用户的详细信息
//		Manager manager = managerService.getByTelephone(currentUsername);
//		if (null != manager) {
//			// //实体类User中包含有用户角色的实体类信息
//			List<ManagerRole> managerRoles = managerService.getManagerRolesByUser(manager);
//			if (null != managerRoles && managerRoles.size() > 0) {
//				// //获取当前登录用户的角色
//				for (ManagerRole managerRole : managerRoles) {
//					int roleid = managerRole.getRoleid();
//					System.out.println("roleid:"+managerRole.getRoleid());
//					List<Permission> permissions = permissionService.getPermissionsByRole(roleid);
//					Role role = roleService.getRoleById(roleid);
//					roleList.add(role.getRolename());
//					for (Permission permission : permissions) {
//						permissionList.add(permission.getPermissionName());
//						System.out.println("pname:"+permission.getPermissionName());
//					}
//				}
//			}
//		}
		// //为当前用户设置角色和权限
		SimpleAuthorizationInfo simpleAuthorInfo = new SimpleAuthorizationInfo();
//		simpleAuthorInfo.addRoles(roleList);
//		simpleAuthorInfo.addStringPermissions(permissionList);
		return simpleAuthorInfo;
	}

	/**
	 * 验证当前登录的Subject 该方法的调用时机为LoginController.login()方法中执行Subject.login()时
	 */
	@Override
	protected AuthenticationInfo doGetAuthenticationInfo(
			AuthenticationToken authcToken) throws AuthenticationException {
//		// 获取基于用户名和密码的令牌
//		// 实际上这个authcToken是从LoginController里面currentUser.login(token)传过来的
//		// 两个token的引用都是一样的
//		UsernamePasswordToken token = (UsernamePasswordToken) authcToken;
//		LoggerUtils.info(getClass().getName(), "token pwd:"+String.valueOf(token.getPassword()));
//		token.setPassword(MD5Util.MD5(String.valueOf(token.getPassword())).toCharArray());
//		System.out.println("验证当前Subject时获取到token为"
//				+ ReflectionToStringBuilder.toString(token,
//						ToStringStyle.MULTI_LINE_STYLE));
//		Manager manager = managerService.getByTelephone(token.getUsername());
//		if (null != manager) {
//			AuthenticationInfo authcInfo = new SimpleAuthenticationInfo(
//					manager.getTelephone(), manager.getPassword(), manager.getNickName());
//			this.setSession("currentUser", manager);
//			return authcInfo;
//		}
		// 此处无需比对,比对的逻辑Shiro会做,我们只需返回一个和令牌相关的正确的验证信息
		// 说白了就是第一个参数填登录用户名,第二个参数填合法的登录密码
		// 这样一来,在随后的登录页面上就只有这里指定的用户和密码才能通过验证
		// 没有返回登录用户名对应的SimpleAuthenticationInfo对象时,就会在LoginController中抛出UnknownAccountException异常
		return null;
	}

	/**
	 * 将一些数据放到ShiroSession中,以便于其它地方使用
	 * 比如Controller,使用时直接用HttpSession.getAttribute(key)就可以取到
	 */
	private void setSession(Object key, Object value) {
		Subject currentUser = SecurityUtils.getSubject();
		if (null != currentUser) {
			Session session = currentUser.getSession();
			System.out
					.println("Session默认超时时间为[" + session.getTimeout() + "]毫秒");
			if (null != session) {
				session.setAttribute(key, value);
			}
		}
	}
}
